absl-py>=1.1.0
apache_beam
clu>=0.0.7
flax==0.3.6
jax==0.2.26
jaxlib==0.1.75
numpy>=1.19.2
sentencepiece>=0.1.95
six>=1.16.0
tensor2tensor>=1.11.0
tensorflow-datasets>=4.3.0
tensorflow>=2.4.1
